# coding=UTF-8
import pymysql
import xlrd

# -*- coding: UTF-8 -*-
#-*-coding:gb2312-*-
'''
    连接数据库  链接暂时注释
    args：db_name（数据库名称）
    returns:db

'''

#db='base', user='root', passwd='pwd', host='localhost', port=XXXX
def mysql_link(de_name):
    try:
        # db = pymysql.connect(host="localhost:3306",
        #                      user="root",
        #                      password="123456",
        #                      database="kiton")

        db = pymysql.connect(host='123.207.159.172', port=3306, user='autotest', passwd='xqNJbv58hpnk', db='autotest',
                               charset='utf8')
        # db = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='123456', db='kiton',
        #                      charset='utf8')
        return db
    except:
        print("could not connect to mysql server")

'''
    读取excel函数
    args：excel_file（excel文件，目录在py文件同目录）
    returns：book
'''
def open_excel(excel_file):
    try:
        book = xlrd.open_workbook(excel_file)  #文件名，把文件与py文件放在同一目录下
        return book
    except:
        print("open excel file failed!")


'''
    执行插入操作
    args:db_name（数据库名称）
         table_name(表名称）
         excel_file（excel文件名，把文件与py文件放在同一目录下）

'''


def store_to(db_name, table_name, excel_file):
    db = mysql_link(db_name)  # 打开数据库连接
    cursor = db.cursor()  # 使用 cursor() 方法创建一个游标对象 cursor

    book = open_excel(excel_file)  # 打开excel文件
    sheets = book.sheet_names()  # 获取所有sheet表名

    #row_data[0]:产品名称；row_data[1]:项目编号；row_data[2]:项目名称；row_data[3]:项目概述；row_data[4]:商机获得方式
    #row_data[5]:需求联系人；row_data[6]:分类；row_data[7]:旧项目阶段；row_data[8]:状态；row_data[9]:master；
    #row_data[10]:分值；row_data[11]:运营经理；row_data[12]:产品经理；row_data[13]:技术经理；row_data[14]:测试经理
    #row_data[15]:项目预算；row_data[16]:备注；row_data[17]:可行性；row_data[18]:范围；row_data[19]:投入产出；row_data[20]:风险分析

    for sheet in sheets:
        sh = book.sheet_by_name(sheet)  # 打开每一张表
        row_num = sh.nrows
        print(row_num)
        list = []  # 定义列表用来存放数据

        for i in range(1, row_num):  # 第一行是标题名，对应表中的字段名所以应该从第二行开始，计算机以0开始计数，所以值是1
            row_data = sh.row_values(i)  # 按行获取excel的值
            #产品名称转换成id
            prd="";
            if (row_data[0]=="信贷体系"):
                prd="1"
            elif (row_data[0]=="数字化运营"):
                prd = "2"
            elif (row_data[0]=="场景化体系"):
                prd = "3"
            elif (row_data[0]=="分行特色"):
                prd = "4"

            # 状态转换成枚举值编码
            status = "";
            if (row_data[8]=="正常"):
                status="00"
            elif (row_data[8]=="提前"):
                status="01"
            if (row_data[8]=="延迟"):
                status="02"
            if (row_data[8]=="暂停"):
                status="03"

            value = (prd, row_data[1], row_data[2], row_data[3], row_data[4], row_data[5], \
                     row_data[6], row_data[7], status, row_data[9], row_data[10], row_data[11], row_data[12],
                     row_data[13], row_data[14], row_data[15], row_data[16], row_data[17], row_data[18], \
                     row_data[19], row_data[20])
            list.append(value)  # 将数据暂存在列表, row_data[14]
            # print(i)
            # row_data[0]:产品名称；row_data[1]:项目编号；row_data[2]:项目名称；row_data[3]:项目概述；row_data[4]:商机获得方式
            # row_data[5]:需求联系人；row_data[6]:分类；row_data[7]:旧项目阶段；row_data[8]:状态；row_data[9]:master；
            # row_data[10]:分值；row_data[11]:运营经理；row_data[12]:产品经理；row_data[13]:技术经理；row_data[14]:测试经理
            # row_data[15]:项目预算；row_data[16]:备注；row_data[17]:可行性；row_data[18]:范围；row_data[19]:投入产出；row_data[20]:风险分析
        sql = "INSERT INTO " + table_name + " ( prd_id,project_id,project_name,summary,origin,\
            introducer,priority,stage_old,status_currentweek,master,score,business_manager,\
            product_manager,develop_manager,test_manager,budget,content,feasibility_analysis,\
            scope,ro_analysis,risk_analysis)VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
        cursor.executemany(sql, list)  # 执行sql语句
        db.commit()  # 提交
        list.clear()  # 清空list
        print("worksheets: " + sheet + " has been inserted " + str(row_num) + " datas!")
    cursor.close()  # 关闭连接
    db.close()


if __name__ == '__main__':
    store_to('pmotool', 'project', '20200525.xlsx')